from langchain import OpenAI, SQLDatabase, SQLDatabaseChain
from langchain.prompts.prompt import PromptTemplate
db = SQLDatabase.from_uri(
    "mysql+pymysql://root:password@115.159.119.116:3306/openai_demo",
    include_tables=['data_wide_s_trade'],
    sample_rows_in_table_info=10
    )
llm = OpenAI(temperature=0)
# 如果您要查询表的多行，您可以使用“top_k”参数选择要获取的最大结果数（默认值为 10）。这对于避免查询结果超出提示最大长度或不必要地消耗令牌非常有用。
db_chain = SQLDatabaseChain.from_llm(llm, db, verbose=True, use_query_checker=True, return_intermediate_steps=True, top_k=10)
result = db_chain("有多少合同?")
print(result["intermediate_steps"])